#!/bin/bash


singularity exec --nv $1 python3 -c "import torch ; \
        print('torch version :',torch.__version__); \
        print('torch version cuda :',torch.version.cuda); \
        print(torch.cuda.is_available()); \
        print(torch.cuda.get_device_name(0)) ; \
        import cupy; \
        print('cupy version',cupy.__version__);\
device = cupy.cuda.Device(0);\
print('cupy.cuda.runtime.runtimeGetVersion :',cupy.cuda.runtime.runtimeGetVersion()) ; \
print('Compute Capability:', device.compute_capability); \
print('Total Memory:', device.mem_info[1]); \
        "
